FROM registry.openanolis.cn/openanolis/cuda:12.1.1-23-runtime

LABEL \
    maintainer="OpenAnolis AI SIG" \
    org.opencontainers.image.title="pytorch" \
    org.opencontainers.image.version="2.0.1-23" \
    org.opencontainers.image.date="20231127"

RUN dnf install pytorch torch-vision -y

CMD ["/bin/bash"]
